--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
tests/backend/test_identity_switch.py e46eed7f7e2637c4c21f4c772d908ef75ec49e6a (e46eed7f) Text, 11.61 KB
T8b949e# SPDX-License-Identifier: 0BSD
Tff7b72import T7ee787json
Tff7b72import T7ee787os
Tff7b72import T7ee787shutil
Tff7b72import T7ee787tempfile
Tff7b72from T7ee787contextlib Tff7b72import Te6edf3ExitStack
Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3AsyncMockTb4b4b4, Te6edf3MagicMockTb4b4b4, Te6edf3patch
Tff7b72import T7ee787pytest
Tff7b72import T7ee787RNS
Tff7b72from T7ee787meshchatxT7ee787.T7ee787meshchat Tff7b72import Te6edf3ReticulumMeshChat
Tf0883e@pytestTff7b72.Td2a8fffixture
Tff7b72def Td2a8fftemp_dirTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3dir_path Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)
Tff7b72yield Te6edf3dir_path
Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Te6edf3dir_pathTb4b4b4)
Tf0883e@pytestTff7b72.Td2a8fffixture
Tff7b72def Td2a8ffmock_rnsTb4b4b4(Tb4b4b4)Tb4b4b4:
T8b949e# Save real Identity class to use as base class for our mock class
Te6edf3real_identity_class Tff7b72= Te6edf3RNSTff7b72.Td2a8ffIdentity
Tff7b72class T56d364MockIdentityClassTb4b4b4(Te6edf3real_identity_classTb4b4b4)Tb4b4b4:
Tff7b72def Tff7b72__init__Tb4b4b4(Tff7b72selfTb4b4b4, Tff7b72*Te6edf3argsTb4b4b4, Tff7b72*Tff7b72*Te6edf3kwargsTb4b4b4)Tb4b4b4:
Tff7b72selfTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffinitial_hash_32_bytes_long_01234Ta5d6ff"
Tff7b72selfTff7b72.Td2a8ffhexhash Tff7b72= Tff7b72selfTff7b72.Td2a8ffhashTff7b72.Td2a8ffhexTb4b4b4(Tb4b4b4)
Tff7b72with Te6edf3ExitStackTb4b4b4(Tb4b4b4) Tff7b72as Te6edf3stackTb4b4b4:
T8b949e# Define patches
Te6edf3patches Tff7b72= Tb4b4b4[
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.ReticulumTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.TransportTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4, Te6edf3MockIdentityClassTb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffthreading.ThreadTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.DatabaseTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.ConfigManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.MessageHandlerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.AnnounceManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.ArchiverManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.MapManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.DocsManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.NomadNetworkManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.TelephoneManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.VoicemailManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.RingtoneManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.RNCPHandlerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.RNStatusHandlerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.RNProbeHandlerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.TranslatorHandlerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.CommunityInterfacesManagerTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffLXMF.LXMRouterTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.IdentityContextTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4]
T8b949e# Apply patches
Te6edf3mocks Tff7b72= Tb4b4b4{Tb4b4b4}
Tff7b72for Te6edf3p Tff7b72in Te6edf3patchesTb4b4b4:
Te6edf3attr_name Tff7b72= Tb4b4b4(
Te6edf3pTff7b72.Td2a8ffattribute Tff7b72if Tffa657hasattrTb4b4b4(Te6edf3pTb4b4b4, Ta5d6ff"Ta5d6ffattributeTa5d6ff"Tb4b4b4) Tff7b72else Te6edf3pTff7b72.Td2a8fftargetTff7b72.Td2a8ffsplitTb4b4b4(Ta5d6ff"Ta5d6ff.Ta5d6ff"Tb4b4b4)Tb4b4b4[Tff7b72-T79c0ff1Tb4b4b4]
Tb4b4b4)
Te6edf3mocksTb4b4b4[Te6edf3attr_nameTb4b4b4] Tff7b72= Te6edf3stackTff7b72.Td2a8ffenter_contextTb4b4b4(Te6edf3pTb4b4b4)
T8b949e# Mock class methods on MockIdentityClass
Te6edf3mock_id_instance Tff7b72= Te6edf3MockIdentityClassTb4b4b4(Tb4b4b4)
Te6edf3mock_id_instanceTff7b72.Td2a8ffget_private_key Tff7b72= Te6edf3MagicMockTb4b4b4(
Te6edf3return_valueTff7b72=Ta5d6ffbTa5d6ff"Ta5d6ffinitial_private_keyTa5d6ff"Tb4b4b4,
Tb4b4b4)
Te6edf3stackTff7b72.Td2a8ffenter_contextTb4b4b4(
Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3MockIdentityClassTb4b4b4, Ta5d6ff"Ta5d6fffrom_fileTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3stackTff7b72.Td2a8ffenter_contextTb4b4b4(
Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3MockIdentityClassTb4b4b4, Ta5d6ff"Ta5d6ffrecallTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3stackTff7b72.Td2a8ffenter_contextTb4b4b4(
Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(
Te6edf3MockIdentityClassTb4b4b4,
Ta5d6ff"Ta5d6fffrom_bytesTa5d6ff"Tb4b4b4,
Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4,
Tb4b4b4)Tb4b4b4,
Tb4b4b4)
T8b949e# Access specifically the ones we need to configure
Te6edf3mock_config Tff7b72= Te6edf3mocksTb4b4b4[Ta5d6ff"Ta5d6ffConfigManagerTa5d6ff"Tb4b4b4]
T8b949e# Setup mock config
Te6edf3mock_configTff7b72.Td2a8ffreturn_valueTff7b72.Td2a8ffdisplay_nameTff7b72.Td2a8ffgetTff7b72.Td2a8ffreturn_value Tff7b72= Ta5d6ff"Ta5d6ffTest UserTa5d6ff"
Tff7b72yield Tb4b4b4{
Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4: Te6edf3MockIdentityClassTb4b4b4,
Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4: Te6edf3mock_id_instanceTb4b4b4,
Ta5d6ff"Ta5d6ffIdentityContextTa5d6ff"Tb4b4b4: Te6edf3mocksTb4b4b4[Ta5d6ff"Ta5d6ffIdentityContextTa5d6ff"Tb4b4b4]Tb4b4b4,
Tb4b4b4}
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_hotswap_identity_successTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:
Te6edf3app Tff7b72= Te6edf3ReticulumMeshChatTb4b4b4(
Te6edf3identityTff7b72=Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4,
Te6edf3storage_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Te6edf3reticulum_config_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Tb4b4b4)
T8b949e# Setup new identity
Te6edf3new_hash Tff7b72= Ta5d6ff"Ta5d6ffnew_hash_123Ta5d6ff"
Te6edf3identity_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3new_hashTb4b4b4)
Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3identity_dirTb4b4b4)
Te6edf3identity_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffwbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffnew_private_keyTa5d6ff"Tb4b4b4)
Te6edf3new_id_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3new_id_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffnew_hash_32_bytes_long_012345678Ta5d6ff"
Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4]Tff7b72.Td2a8fffrom_fileTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3new_id_instance
T8b949e# Configure mock context
Te6edf3mock_context Tff7b72= Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffIdentityContextTa5d6ff"Tb4b4b4]Tff7b72.Td2a8ffreturn_value
Te6edf3mock_contextTff7b72.Td2a8ffconfigTff7b72.Td2a8ffdisplay_nameTff7b72.Td2a8ffgetTff7b72.Td2a8ffreturn_value Tff7b72= Ta5d6ff"Ta5d6ffNew UserTa5d6ff"
Te6edf3mock_contextTff7b72.Td2a8ffidentity_hash Tff7b72= Te6edf3new_hash
T8b949e# Mock methods
Te6edf3appTff7b72.Td2a8ffteardown_identity Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffsetup_identity Tff7b72= Te6edf3MagicMockTb4b4b4(
Te6edf3side_effectTff7b72=Tff7b72lambda Tffa657idTb4b4b4: Tffa657setattrTb4b4b4(Te6edf3appTb4b4b4, Ta5d6ff"Ta5d6ffcurrent_contextTa5d6ff"Tb4b4b4, Te6edf3mock_contextTb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffwebsocket_broadcast Tff7b72= Te6edf3AsyncMockTb4b4b4(Tb4b4b4)
T8b949e# Perform hotswap
Te6edf3result Tff7b72= Tff7b72await Te6edf3appTff7b72.Td2a8ffhotswap_identityTb4b4b4(Te6edf3new_hashTb4b4b4)
Tff7b72assert Te6edf3result Tff7b72is Tff7b72True
Te6edf3appTff7b72.Td2a8ffteardown_identityTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffsetup_identityTff7b72.Td2a8ffassert_called_once_withTb4b4b4(Te6edf3new_id_instanceTb4b4b4)
Te6edf3appTff7b72.Td2a8ffwebsocket_broadcastTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)
Te6edf3ws_payload Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3appTff7b72.Td2a8ffwebsocket_broadcastTff7b72.Td2a8ffcall_argsTb4b4b4[T79c0ff0Tb4b4b4]Tb4b4b4[T79c0ff0Tb4b4b4]Tb4b4b4)
Tff7b72assert Te6edf3ws_payloadTb4b4b4[Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffidentity_switchedTa5d6ff"
Tff7b72assert Te6edf3ws_payloadTb4b4b4[Ta5d6ff"Ta5d6ffidentity_hashTa5d6ff"Tb4b4b4] Tff7b72== Te6edf3new_hash
Tff7b72assert Te6edf3ws_payloadTb4b4b4[Ta5d6ff"Ta5d6ffdisplay_nameTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffNew UserTa5d6ff"
T8b949e# Verify main identity file was updated
Te6edf3main_identity_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3main_identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffrbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Tff7b72assert Te6edf3fTff7b72.Td2a8ffreadTb4b4b4(Tb4b4b4) Tff7b72== Ta5d6ffbTa5d6ff"Ta5d6ffnew_private_keyTa5d6ff"
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_hotswap_identity_keep_aliveTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:
Te6edf3app Tff7b72= Te6edf3ReticulumMeshChatTb4b4b4(
Te6edf3identityTff7b72=Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4,
Te6edf3storage_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Te6edf3reticulum_config_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Tb4b4b4)
T8b949e# Setup new identity
Te6edf3new_hash Tff7b72= Ta5d6ff"Ta5d6ffnew_hash_123Ta5d6ff"
Te6edf3identity_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3new_hashTb4b4b4)
Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3identity_dirTb4b4b4)
Te6edf3identity_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffwbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffnew_private_keyTa5d6ff"Tb4b4b4)
Te6edf3new_id_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3new_id_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffnew_hash_32_bytes_long_012345678Ta5d6ff"
Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4]Tff7b72.Td2a8fffrom_fileTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3new_id_instance
T8b949e# Configure mock context
Te6edf3mock_context Tff7b72= Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffIdentityContextTa5d6ff"Tb4b4b4]Tff7b72.Td2a8ffreturn_value
Te6edf3mock_contextTff7b72.Td2a8ffconfigTff7b72.Td2a8ffdisplay_nameTff7b72.Td2a8ffgetTff7b72.Td2a8ffreturn_value Tff7b72= Ta5d6ff"Ta5d6ffNew UserTa5d6ff"
Te6edf3mock_contextTff7b72.Td2a8ffidentity_hash Tff7b72= Te6edf3new_hash
T8b949e# Mock methods
Te6edf3appTff7b72.Td2a8ffteardown_identity Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffsetup_identity Tff7b72= Te6edf3MagicMockTb4b4b4(
Te6edf3side_effectTff7b72=Tff7b72lambda Tffa657idTb4b4b4: Tffa657setattrTb4b4b4(Te6edf3appTb4b4b4, Ta5d6ff"Ta5d6ffcurrent_contextTa5d6ff"Tb4b4b4, Te6edf3mock_contextTb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffwebsocket_broadcast Tff7b72= Te6edf3AsyncMockTb4b4b4(Tb4b4b4)
T8b949e# Perform hotswap with keep_alive=True
Te6edf3result Tff7b72= Tff7b72await Te6edf3appTff7b72.Td2a8ffhotswap_identityTb4b4b4(Te6edf3new_hashTb4b4b4, Te6edf3keep_aliveTff7b72=Tff7b72TrueTb4b4b4)
Tff7b72assert Te6edf3result Tff7b72is Tff7b72True
Te6edf3appTff7b72.Td2a8ffteardown_identityTff7b72.Td2a8ffassert_not_calledTb4b4b4(Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffsetup_identityTff7b72.Td2a8ffassert_called_once_withTb4b4b4(Te6edf3new_id_instanceTb4b4b4)
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_hotswap_identity_file_missingTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:
Te6edf3app Tff7b72= Te6edf3ReticulumMeshChatTb4b4b4(
Te6edf3identityTff7b72=Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4,
Te6edf3storage_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Te6edf3reticulum_config_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Tb4b4b4)
T8b949e# Attempt hotswap with non-existent hash
Te6edf3result Tff7b72= Tff7b72await Te6edf3appTff7b72.Td2a8ffhotswap_identityTb4b4b4(Ta5d6ff"Ta5d6ffnon_existent_hashTa5d6ff"Tb4b4b4)
Tff7b72assert Te6edf3result Tff7b72is Tff7b72False
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_hotswap_identity_corruptedTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:
Te6edf3app Tff7b72= Te6edf3ReticulumMeshChatTb4b4b4(
Te6edf3identityTff7b72=Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4,
Te6edf3storage_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Te6edf3reticulum_config_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Tb4b4b4)
T8b949e# Setup "corrupted" identity
Te6edf3new_hash Tff7b72= Ta5d6ff"Ta5d6ffcorrupted_hashTa5d6ff"
Te6edf3identity_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3new_hashTb4b4b4)
Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3identity_dirTb4b4b4)
Te6edf3identity_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffwbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffcorrupted_dataTa5d6ff"Tb4b4b4)
Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4]Tff7b72.Td2a8fffrom_fileTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72None
T8b949e# Perform hotswap
Te6edf3result Tff7b72= Tff7b72await Te6edf3appTff7b72.Td2a8ffhotswap_identityTb4b4b4(Te6edf3new_hashTb4b4b4)
Tff7b72assert Te6edf3result Tff7b72is Tff7b72False
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_hotswap_identity_recoveryTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:
Te6edf3app Tff7b72= Te6edf3ReticulumMeshChatTb4b4b4(
Te6edf3identityTff7b72=Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4,
Te6edf3storage_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Te6edf3reticulum_config_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Tb4b4b4)
T8b949e# Save initial identity file
Te6edf3main_identity_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3main_identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffwbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffinitial_private_keyTa5d6ff"Tb4b4b4)
T8b949e# Setup new identity
Te6edf3new_hash Tff7b72= Ta5d6ff"Ta5d6ffnew_hash_123Ta5d6ff"
Te6edf3identity_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3new_hashTb4b4b4)
Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3identity_dirTb4b4b4)
Te6edf3identity_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffwbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffnew_private_keyTa5d6ff"Tb4b4b4)
Te6edf3new_id_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3new_id_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffnew_hash_32_bytes_long_012345678Ta5d6ff"
Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4]Tff7b72.Td2a8fffrom_fileTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3new_id_instance
T8b949e# Mock setup_identity to fail first time (after hotswap start),
T8b949e# but the second call (recovery) should succeed.
Te6edf3appTff7b72.Td2a8ffsetup_identity Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3side_effectTff7b72=Tb4b4b4[Tf85149ExceptionTb4b4b4(Ta5d6ff"Ta5d6ffSetup failedTa5d6ff"Tb4b4b4)Tb4b4b4, Tff7b72NoneTb4b4b4]Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffteardown_identity Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffwebsocket_broadcast Tff7b72= Te6edf3AsyncMockTb4b4b4(Tb4b4b4)
T8b949e# Perform hotswap
Te6edf3result Tff7b72= Tff7b72await Te6edf3appTff7b72.Td2a8ffhotswap_identityTb4b4b4(Te6edf3new_hashTb4b4b4)
Tff7b72assert Te6edf3result Tff7b72is Tff7b72False
Tff7b72assert Te6edf3appTff7b72.Td2a8ffsetup_identityTff7b72.Td2a8ffcall_count Tff7b72== T79c0ff2
T8b949e# Verify main identity file was restored
Tff7b72with Tffa657openTb4b4b4(Te6edf3main_identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffrbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Tff7b72assert Te6edf3fTff7b72.Td2a8ffreadTb4b4b4(Tb4b4b4) Tff7b72== Ta5d6ffbTa5d6ff"Ta5d6ffinitial_private_keyTa5d6ff"
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_hotswap_identity_ultimate_failure_emergency_identityTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:
Te6edf3app Tff7b72= Te6edf3ReticulumMeshChatTb4b4b4(
Te6edf3identityTff7b72=Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4,
Te6edf3storage_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Te6edf3reticulum_config_dirTff7b72=Te6edf3temp_dirTb4b4b4,
Tb4b4b4)
T8b949e# Setup new identity
Te6edf3new_hash Tff7b72= Ta5d6ff"Ta5d6ffnew_hash_123Ta5d6ff"
Te6edf3identity_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3new_hashTb4b4b4)
Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3identity_dirTb4b4b4)
Te6edf3identity_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffwbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffnew_private_keyTa5d6ff"Tb4b4b4)
Te6edf3new_id_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3new_id_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffnew_hash_32_bytes_long_012345678Ta5d6ff"
Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4]Tff7b72.Td2a8fffrom_fileTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3new_id_instance
T8b949e# Mock setup_identity to fail ALL THE TIME
Te6edf3appTff7b72.Td2a8ffsetup_identity Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3side_effectTff7b72=Tf85149ExceptionTb4b4b4(Ta5d6ff"Ta5d6ffUltimate failureTa5d6ff"Tb4b4b4)Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffteardown_identity Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffwebsocket_broadcast Tff7b72= Te6edf3AsyncMockTb4b4b4(Tb4b4b4)
T8b949e# Mock create_identity to return a new hash
Te6edf3emergency_hash Tff7b72= Ta5d6ff"Ta5d6ffemergency_hash_456Ta5d6ff"
Te6edf3appTff7b72.Td2a8ffcreate_identity Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3return_valueTff7b72=Tb4b4b4{Ta5d6ff"Ta5d6ffhashTa5d6ff"Tb4b4b4: Te6edf3emergency_hashTb4b4b4}Tb4b4b4)
T8b949e# Mock RNS.Identity.from_file for the emergency identity
Te6edf3emergency_id Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3emergency_idTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffemergency_hash_32_bytes_long_012Ta5d6ff"
T8b949e# Ensure from_file returns the new identity when called for the emergency one
Tff7b72def Td2a8ffside_effect_from_fileTb4b4b4(Te6edf3pathTb4b4b4)Tb4b4b4:
Tff7b72if Te6edf3emergency_hash Tff7b72in Te6edf3pathTb4b4b4:
Tff7b72return Te6edf3emergency_id
Tff7b72return Te6edf3new_id_instance
Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4]Tff7b72.Td2a8fffrom_fileTff7b72.Td2a8ffside_effect Tff7b72= Te6edf3side_effect_from_file
T8b949e# Create the directory structure create_identity would have created
Te6edf3emergency_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3emergency_hashTb4b4b4)
Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3emergency_dirTb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3emergency_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)Tb4b4b4, Ta5d6ff"Ta5d6ffwbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffemergency_private_keyTa5d6ff"Tb4b4b4)
T8b949e# Perform hotswap
Te6edf3result Tff7b72= Tff7b72await Te6edf3appTff7b72.Td2a8ffhotswap_identityTb4b4b4(Te6edf3new_hashTb4b4b4)
Tff7b72assert Te6edf3result Tff7b72is Tff7b72False
T8b949e# Should have tried to setup identity 3 times:
T8b949e# 1. new_identity
T8b949e# 2. old_identity (recovery)
T8b949e# 3. emergency_identity (failsafe)
Tff7b72assert Te6edf3appTff7b72.Td2a8ffsetup_identityTff7b72.Td2a8ffcall_count Tff7b72== T79c0ff3
Te6edf3appTff7b72.Td2a8ffcreate_identityTff7b72.Td2a8ffassert_called_once_withTb4b4b4(Te6edf3display_nameTff7b72=Ta5d6ff"Ta5d6ffEmergency RecoveryTa5d6ff"Tb4b4b4)
T8b949e# Verify main identity file was updated to emergency one
Te6edf3main_identity_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)
Tff7b72with Tffa657openTb4b4b4(Te6edf3main_identity_fileTb4b4b4, Ta5d6ff"Ta5d6ffrbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:
Tff7b72assert Te6edf3fTff7b72.Td2a8ffreadTb4b4b4(Tb4b4b4) Tff7b72== Ta5d6ffbTa5d6ff"Ta5d6ffemergency_private_keyTa5d6ff"
──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────